// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "net/http/http_status_code.h"
#include "net/server/http_server_response_info.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace net {

TEST(HttpServerResponseInfoTest, StatusLine)
{
    HttpServerResponseInfo response;
    ASSERT_EQ(HTTP_OK, response.status_code());
    ASSERT_EQ("HTTP/1.1 200 OK\r\n\r\n", response.Serialize());
}

TEST(HttpServerResponseInfoTest, Headers)
{
    HttpServerResponseInfo response;
    response.AddHeader("A", "1");
    response.AddHeader("A", "2");
    ASSERT_EQ("HTTP/1.1 200 OK\r\nA:1\r\nA:2\r\n\r\n", response.Serialize());
}

TEST(HttpServerResponseInfoTest, Body)
{
    HttpServerResponseInfo response;
    ASSERT_EQ(std::string(), response.body());
    response.SetBody("body", "type");
    ASSERT_EQ("body", response.body());
    ASSERT_EQ(
        "HTTP/1.1 200 OK\r\nContent-Length:4\r\nContent-Type:type\r\n\r\nbody",
        response.Serialize());
}

TEST(HttpServerResponseInfoTest, CreateFor404)
{
    HttpServerResponseInfo response = HttpServerResponseInfo::CreateFor404();
    ASSERT_EQ(
        "HTTP/1.1 404 Not Found\r\n"
        "Content-Length:0\r\nContent-Type:text/html\r\n\r\n",
        response.Serialize());
}

TEST(HttpServerResponseInfoTest, CreateFor500)
{
    HttpServerResponseInfo response = HttpServerResponseInfo::CreateFor500("mess");
    ASSERT_EQ(
        "HTTP/1.1 500 Internal Server Error\r\n"
        "Content-Length:4\r\nContent-Type:text/html\r\n\r\nmess",
        response.Serialize());
}

} // namespace net
